当 KL 正则化在群体策略优化中失灵时
🧭 RLHF/GRPO 训练稳定性优化
Alibaba Group
系统分析 GRPO 中参考 KL 与奖励交互的七种失效模式,提出条件 KL 标定组内奖励系数的 ZCPO
📌 工作参考价值(4 点)
- 在电商/广告/Agent 场景用 GRPO 类方法做 LLM 策略优化时,若显式 reference KL 导致奖励收敛慢或指标波动,可尝试去掉 KL 项或改用条件 KL 只用在校准组内奖励系数,而非独立正则项。
- 对于文案生成、搜索词推荐等变长输出任务,注意 KL 会随 response length 增长且可能集中在少数 token,造成长度偏置;可考虑 per-token 或条件归一化,避免长文本被过度惩罚。
- 当组内样本 reward 相同或梯度取消时,独立 KL 梯度不会完全抵消,仍会推动策略漂移;可在组内 reward 方差接近 0 时关闭 KL 更新,或采用 ZCPO 式相对漂移标定。
- 将 KL 估计直接加入 reward 会引入采样噪声,建议分离 KL 估计与 reward 计算,并使用更稳健的组内相对校准系数。
自教师与学生同步:联合在策略学习与教学
🧭 LLM 强化学习 · 自蒸馏训练 ★ 深度解读
Perplexity · NVIDIA
提出JOLT联合训练特权教师与无特权学生,使蒸馏梯度与奖励梯度一致,提升样本效率和最终性能
📌 工作参考价值(4 点)
- 在训练LLM-based推荐/搜索/Agent模型时,若存在特权信息(如用户真实反馈、人工标注、高质量参考),可采用JOLT模式:同一模型联合训练特权教师和无特权学生,教师用KL正则化向学生,避免教师走捷径导致蒸馏信号与奖励梯度相悖。
- 对于稀疏奖励的长程任务(如多轮对话推荐、会话转化优化),JOLT的密集token级监督可大幅提高样本效率;教师可接收用户反馈摘要、历史成功策略等作为privileged context。
- 工程实现简单:共享参数的教师和学生分别生成轨迹,使用logprob ratio作为token权重,结合group advantage归一化,无需额外教师模型。
- 超参数经验:教师KL β_T 和学生蒸馏 β_S 宜设较小值(如β_T=0.001~0.03, β_S=0.1),过大会训练不稳定;学生奖励可用时,JOLT+补充直接RL可进一步提升。
基于比较预言机的 LLM 偏好对齐零阶优化范式
🧭 LLM 偏好对齐 · Zeroth-order 优化 ★ 深度解读
UC Berkeley · New York University · Alibaba DAMO Academy · Columbia University
提出 ComPO,用比较预言机从低 margin/noisy 偏好对中提取方向信号,缓解 likelihood displacement
📌 工作参考价值(4 点)
- 在 LLM 排序、Agent 策略或生成式推荐的 preference tuning 中,不要直接丢弃低 margin 偏好对:按参考策略 logπ 差异切分 clean/noisy,对 noisy 对改用比较信号更新,能保留数据并减少 likelihood displacement。
- 工程上可采用冻结主干、只对 lm_head 输出层做稀疏扰动更新;阈值为保留约 1%-6% 的非零条目,显存可压到 16-30GB 级(A40),适合在已有线上 checkpoint 上做轻量任务适配。
- 在线正则化可参考:用当前 policy 对无标注 prompt 采样,计算 length-normalized log ratio 作为 reverse-KL 代理,soft damping 调步长;配合 successful-batch replay buffer 可进一步稳定 LC win rate。
- 训练诊断不要只看 benchmark:观察 comparison oracle 负输出计数分布、pairwise log-likelihood 变化方向,可作为 likelihood displacement 的早期 sanity check。
SimpleOPD:简单与分词器无关的长上下文推理在线策略蒸馏
🧭 On-policy 蒸馏 · 跨 tokenizer 长上下文推理 ★ 深度解读
Shanghai Artificial Intelligence Laboratory
用共享文本空间跨 tokenizer 对齐并加入终止 token 掩码与参考 KL 稳定长上下文推理蒸馏
📌 工作参考价值(4 点)
- 跨模型/跨 tokenizer 蒸馏工程:业务中想把大推理模型蒸馏到小线上模型,不必做人工 token 映射,直接让 teacher 用自己的 tokenizer 读 student 生成的文本,只对齐文本 span 完全相同的 token;实验中 lexical overlap 始终 >90%,大部分监督信号可以保留,适合 Qwen/Llama/GLM/Gemma 等不同家族的在线蒸馏。
- 长链推理蒸馏的稳定性 trick:OPD 直接从长上下文老师搬推理能力,学生容易出现长度爆炸、截断、重复;可以照搬两个简单做法:① 掩盖 </think>、<|im_end|> 等终止/结构 token 的 advantage,不强制学生模仿老师的终止倾向;② 加 student-reference KL,防止策略偏移。Intern-S2 和 Qwen3.5-35B 上能有效把截断率降到接近 0。
- KL 系数是重要的超参:师生差距越大,越需要更强的 reference KL;GLM-4.7 消融显示 0.5 不足、1.2 过度,1.0 最均衡。业务上做蒸馏可以优先按模型家族差异设 0.5/1.0 两档。
- 训练数据聚焦更利于能力迁移:仅用 proof-only 数据蒸馏,目标证明能力提升最大;混入通用可验证数学数据反而削弱 ProofBench 迁移。类比到业务:要蒸馏商品知识、query 改写或策略推理,训练集应保持任务纯净,不要混太多无关任务。
Se-DPO:自演化令牌信用用于直接偏好优化
🧭 偏好对齐 · 令牌级自适应KL正则化 ★ 深度解读
University of California, Los Angeles · Shanghai AI Laboratory
通过在线从模型自身隐式奖励与参考熵中动态学习每令牌KL预算,解决DPO均匀加权的次优问题。
📌 工作参考价值(4 点)
- 动态重要性加权思想可迁移至推荐场景:类似给用户行为序列中不同位置 token 分配不同 KL 预算,推荐系统中可对用户历史行为的不同步数或不同特征维度分配不同的正则化强度或学习权重,避免平均主义,让关键决策点有更大优化自由度。
- 在线重要性更新机制:静态重要度在训练后期会严重失配,Se-DPO 定期从模型当前状态更新信用。这启发在在线学习排序模型或生成式推荐中,可引入周期性重估每个样本/特征重要性的机制,而不是训练前一次性固定。
- 轻量校准网络的设计模式:用一个小 MLP 融合强度(\(|\hat{r}_t|\))与不确定性(\(H_{\text{ref}}\))两个信号,输出信用。推荐系统可类似设计一个轻量子网络,结合预估的重要度与置信度来动态调节损失权重,几乎无额外成本。
- 无需外部信号的在线信用提取:只用模型自身隐式奖励和参考熵,无需教师模型或额外标注。在缺乏外部监督的生成式推荐或 Agent 任务中,可仿照此方法利用模型内部信号进行自适应样本加权。
不完整图谱证据下的LLM推理接地理论
🧭 LLM推理接地 · 不完整图谱证据
Tianjin Normal University · Harbin Institute of Technology
证明不完整图谱下硬约束无法同时保证拒假和保真,提出KL正则化软接地框架并给出稳定性边界
📌 工作参考价值(4 点)
- 电商知识图谱的软接地:商品关系图谱常不完备,应避免硬性剪枝。采用KL正则化软约束,保留未被观察但非矛盾的推理路径,防止丢弃正确的长尾关联。
- GraphRAG检索策略:在检索增强生成中,不必强制LLM完全忠实于检索到的子图。允许有限松弛可提高对缺失边的鲁棒性,避免因召回不足而拒真。
- Agent多跳推理的稳定性设计:当Agent基于动态图证据进行多步推理时,可利用稳定性边界评估证据扰动的影响,指导高频更新下的容错机制。
- 区分事实与声明:将图谱相容性视为“声明支持”而非事实真理,在广告文案生成或推荐解释中,可避免因信源不完整而误判正确输出。